package com.example.utils;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * TODO
 * Token工具类，用于生成token
 * 用户登录拿到Token才能服务系统资源
 * @Comment：最难的一次作业
 * @Date: 2023/11/19
 */
@Component
public class TokenUtils {
    // 在代码中获取配置文件的值
    @Value("${jwt.secret}")
    private String secret;

    // 在代码中获取配置文件的值
    @Value("${jwt.expiration}")
    private long expiration;

    //根据负载信息获取token
    private String generateJWT(Map<String, Object> map){
                //builder方法构建JWT并生成 JWT 字符串
        return Jwts.builder()
                //设置载荷部分的方法,包含与用户身份验证和授权相关的信息
                .setClaims(map)
                //设置签名的方法，参数：签名算法和密钥
                .signWith(SignatureAlgorithm.HS512,secret)
                //设置过期时间
                .setExpiration(new Date(System.currentTimeMillis() + expiration * 1000))
                //将 JWT 编码为紧凑的字符串形式
                .compact();
    }
    //传入用户登录信息，生成token
    public String generateToken(UserDetails details){
        Map<String, Object> map = new HashMap<>(2);
        map.put("username", details.getUsername());
        //token创建时间
        map.put("created",new Date());
        return this.generateJWT(map);
    }

    /**
     * 根据token获取载荷信息
     * 载荷是一个JSON 对象，它包含了一些与用户身份验证和授权相关的信息，例如用户ID、用户名、角色、权限等
     */

    public Claims getTokenBody(String token){
        //token过期时进行异常捕获
        try {
            //parser方法解析 JWT 字符串
            return Jwts.parser()
                    //设置 JWT 签名密钥的方法,用配置的secret进行加密
                    .setSigningKey(secret)
                    //解析 JWT Token 并获取其中的载荷信息（payload），同时验证签名是否正确
                    .parseClaimsJws(token)
                    //从 JWT 中获取载荷（payload）信息
                    .getBody();
        }   catch (Exception e){
            return null;
        }
    }

    //根据token获取用户名
    public String getUsernameByToken(String token){
        return (String) this.getTokenBody(token).get("username");
    }

    //根据token判断当前时间内，改token是否过期
    public boolean isExpiration(String token){
        return this.getTokenBody(token).getExpiration().before(new Date());
    }

    //刷新token
    public String refreshToken(String token){
        Claims claims = this.getTokenBody(token);
        claims.setExpiration(new Date());
        return this.generateJWT(claims);
    }
}
